基于face_recognition实现人脸识别
作者：强哥，现供职于一家大型全球电子商务网站，多年Python程序员，热爱数据，热爱AI，希望能与更多同业人交流。
个人微信公众号：Python与数据分析
前文传送门：50行代码实现人脸检测
上一篇中我们实现了检测照片中的人脸并标出人脸特征点（鼻子，眼睛，眉毛等），这一篇我们将在上一篇的基础上，进一步实现人脸识别，告诉你照片里的人是谁。
准备工作
我们的人脸识别基于face_recognition库。face_recognition基于dlib实现，用深度学习训练数据，模型准确率高达99.38%。在开始我们的工作前，我们先安装face_recognition
pip install face_recognition
人脸数字化
人脸识别的第一步是检测照片中的人脸区域，然后将人脸的图像数据转换成一个长度为128的向量，这128个数据代表了人脸的128个特征指标，如下所示
对于每一张已知人脸，生成这样的一个128位的向量。对于一张未知人脸，将它的128位向量和所有已知人脸的128位向量一一比对，找到相似度最高的那一个，即找出了未知人脸对应的人。
图片数据
我们准备了两张照片，分别是凯特王妃和威廉王子的单人照，分别存成catherine.jpg和william.jpg，这两张照片中的人脸作为我们的已知人脸
我们的目标是在下面的合影中识别出两人的脸并在图中标出各自的名字。下图存成unknown.jpg
代码实现
接下来开始我们的编程工作
import cv2
import face_recognitionnames = [
"catherine",
"william",
]
首先我们定义了标签集，存在names数组中。
标签名字也是我们图片的文件名。
images = []
for name in names:
filename = name + ".jpg"
image = face_recognition.load_image_file(filename)
images.append(image)
unknown_image = face_recognition.load_image_file("unknown.jpg")
调用face_recognition.load_image_file从图片中读取数据。
这里读取了包含已知人脸和未知人脸的图片的数据，未知人脸的图片就是上面的合影图片unknown.jpg。
face_encodings = []
for image in images:
encoding = face_recognition.face_encodings(image)[0]
face_encodings.append(encoding)
unknown_face_encodings = face_recognition.face_encodings(unknown_image)
face_recognition.face_encodings会返回图片中的所有的人脸的128位向量。单人照片只有一张人脸，所以face_recognition.face_encodings(image)[0]只取第一个元素。合影图片中包含了2张人脸，所以unknown_face_encodings包含2个128位向量。
face_locations = face_recognition.face_locations(unknown_image)
for i in range(len(unknown_face_encodings)):
unknown_encoding = unknown_face_encodings[i]
face_location = face_locations[i]
top, right, bottom, left = face_location
cv2.rectangle(unknown_image, (left, top), (right, bottom), (0, 255, 0), 2)
results = face_recognition.compare_faces(face_encodings, unknown_encoding)
for j in range(len(results)):
if results[j]:
name = names[j]
cv2.putText(unknown_image, name, (left-10, top-10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
unknown_image_rgb = cv2.cvtColor(unknown_image, cv2.COLOR_BGR2RGB)
cv2.imshow("Output", unknown_image_rgb)
cv2.waitKey(0)
face_locations存了每张脸的位置信息。
在循环中我们调用cv2.rectangle框出了检测到的每张脸。
face_recognition.compare_faces将已知人脸的128位向量和每张未知人脸的128位向量做比较，结果存入results数组中。results数组中的每一个元素都是True或者False，长度和人脸个数相等。results中的每个元素都和已知人脸一一对应，在某一个位置处的元素为True，表示未知人脸被识别成这张已知人脸。
对识别出来的每张人脸，我们调用cv2.putText在图上标注标签。
以上是代码的全部。
测试
令人兴奋的时刻又来到了！ 我们来检验一下我们的成果。
运行上面的程序，可以看到下面的结果
威廉王子和凯特王妃的人脸都被准确地识别出来。绿色的框框出了人脸区域，框的上方标注了识别到的人的名字。
至此，我们成功地实现了人脸识别。
后续不定期的更新一些图像处理方面其他有趣的应用，例如抠图、实现手绘效果、二维码识别、验证码识别，等等。
Python爱好者社区历史文章大合集：
Python爱好者社区历史文章列表（每周append更新一次）
关注后在公众号内回复“课程”即可获取：
小编的Python入门视频课程！！！
崔老师爬虫实战案例免费学习视频。
丘老师数据科学入门指导免费学习视频。
陈老师数据分析报告制作免费学习视频。
玩转大数据分析！Spark2.X+Python 精华实战课程免费学习视频。
丘老师Python网络爬虫实战免费学习视频。